import numpy as np

# 文件路径列表
npz_files = [
    'samples/SiT-B-2-0040000-size-256-vae-ema-cfg-1.8-seed-0-sde.npz',
    'samples/ref/cifar10.npz'
]

# 遍历每个文件
for ref in npz_files:
    print(f"Loading file: {ref}")
    npz_file = np.load(ref)
    print("Keys in the .npz file:", npz_file.files)
    for key in npz_file.files:
        print(f"Shape of '{key}':", npz_file[key].shape)
    print("-" * 40)  # 分隔线